Skip to content

Conversation

AntoineRichard
Copy link

Hey there!

This is a simple PR that brings a bit more flexibility when using a WandB logger.
It's following the same logic as SKRL's implementation.

I.e. instead of passing:

wandb_project = "Whatever"

We can now use:

 wandb_kwargs = {
        "project": "jetbot_direct",
        "entity": "spacer-rl",
        "group": "zeroG",
    } 

I'm happy to make any modification, like making it backward compatible with the current default config style.

LMK,

Antoine

@AntoineRichard AntoineRichard changed the title Changed the way WandB is being initialized. It should be more flexible. More Flexible WandB Interface Apr 1, 2025
@@ -25,17 +26,19 @@ def __init__(self, log_dir: str, flush_secs: int, cfg):
run_name = os.path.split(log_dir)[-1]

try:
project = cfg["wandb_project"]
project = cfg["wandb_kwargs"]["project"]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could you also modify the example_config.yaml to reflect the change?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants